How to solve functions in c hacker rank